package hot

import (
	"sort"
)

/**
原题链接:
https://leetcode.cn/problems/task-scheduler/description/

 * hot 621.任务调度器
 * 给你一个用字符数组 tasks 表示的 CPU 需要执行的任务列表。
 * 其中每个字母表示一种不同种类的任务。
 * 任务可以以任意顺序执行，并且每个任务都可以在 1 个单位时间内执行完。
 * 在任何一个单位时间，CPU 可以完成一个任务，或者处于待命状态。
 *
 * 然而，两个 相同种类 的任务之间必须有长度为整数 n 的冷却时间，
 * 因此至少有连续 n 个单位时间内 CPU 在执行不同的任务，或者在待命状态。
 * 你需要计算完成所有任务所需要的 最短时间 。



任务调度器:
给你一个用字符数组 tasks表示的 cpu需要执行的任务列表.
其中每个字母表示一种不同种类的任务.
任务可以以任意顺序执行, 并且每个任务都可以在 1个单位时间内执行完.
在任何一个单位时间, cpu可以完成一个任务, 或者处于待命状态.

然而, 两个 相同种类的任务之间必须有长度为整数 n的冷却时间.
因此至少有连续 n个单位时间内 cpu
 *
 * 示例 1：
 * 输入：tasks = ["A","A","A","B","B","B"], n = 2
 * 输出：8
 * 解释：A -> B -> (待命) -> A -> B -> (待命) -> A -> B
 *      在本示例中，两个相同类型任务之间必须间隔长度为 n = 2 的冷却时间，
 *      而执行一个任务只需要一个单位时间，所以中间出现了（待命）状态。

 *
 * 示例 2：
 * 输入：tasks = ["A","A","A","B","B","B"], n = 0
 * 输出：6
 * 解释：在这种情况下，任何大小为 6 的排列都可以满足要求，因为 n = 0
 * ["A","A","A","B","B","B"]
 * ["A","B","A","B","A","B"]
 * ["B","B","B","A","A","A"]
 * ...
 * 诸如此类
 *
 *
 * 示例 3：
 * 输入：tasks = ["A","A","A","A","A","A","B","C","D","E","F","G"], n = 2
 * 输出：16
 * 解释：一种可能的解决方案是：
 *      A -> B -> C -> A -> D -> E -> A -> F -> G -> A -> (待命) -> (待命) -> A -> (待命) -> (待命) -> A
*/

//一刷

func leastInterval(tasks []byte, n int) int {

	//1.先统计各个任务出现的次数
	count := make([]int, 26)

	for _, task := range tasks {
		count[task-'A']++
	}

	//2.根据出现次数, 进行排序
	sort.Ints(count)

	//3.找出有多少个和最多出现次数一样的任务
	maxCount := 0
	for i := 25; i >= 0; i-- {
		if count[i] != count[25] {
			break
		}
		maxCount++
	}

	//4.返回结果  len(tasks)表示特殊情况:
	//每个任务都出现一次, 并且间隔时间为0,那么len(tasks)的长度才是结果
	//例如: ["A","A","A","B","B","B"]  并且n=0   那么公式为4,但实际结果应该为6
	return max((count[25]-1)*(n+1)+maxCount, len(tasks))
}

func max(a int, b int) int {
	if a > b {
		return a
	}

	return b
}
